// Copyright (C) Kumo inc. and its affiliates.
// Author: Jeff.li lijippy@163.com
// All rights reserved.
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
//

#include <cantor/parser/ast/func_ctor.h>

namespace cantor::parser {
    nlohmann::ordered_json CtorFuncExpr::to_json_impl() const {
        auto j = std::move(DispatchExpr::to_json_impl());
        KCHECK(!fn_name.empty()) << "not set fn_name by:" << pb::ast::ExprNodeType_Name(expr_type);
        j["fn_name"] = fn_name.to_string();
        return j;
    }

    nlohmann::ordered_json CtorMapFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorStructFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorUnionFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        if (arg) {

        } else {
            j["arg"] = nlohmann::ordered_json();
        }
        return j;
    }

    nlohmann::ordered_json CtorArrayFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorSetFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorTupleFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorJsonFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();

        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorJsonArrayFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorEnumFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorProtoFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }

    nlohmann::ordered_json CtorUdfFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        j["arg"] = arg.to_json();
        return j;
    }


    nlohmann::ordered_json CtorDefaultFuncExpr::to_json() const {
        auto j = CtorFuncExpr::to_json_impl();
        if (arg) {
            j["arg"] = arg->to_json();
        } else {
            j["arg"] = nlohmann::ordered_json();
        }
        return j;
    }
} // namespace cantor::parser
